package com.zjj.algorithm.learning.dmsxl.backtrack;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;

/**
 * 77. 组合 中档题
 * <p>
 * 给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
 * <p>
 * 你可以按 任何顺序 返回答案。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：n = 4, k = 2
 * 输出：
 * [
 * [2,4],
 * [3,4],
 * [2,3],
 * [1,2],
 * [1,3],
 * [1,4],
 * ]
 * 示例 2：
 * <p>
 * 输入：n = 1, k = 1
 * 输出：[[1]]
 * <p>
 * <p>
 * 提示：
 * <p>
 * 1 <= n <= 20
 * 1 <= k <= n
 *
 * @author zjj_admin
 * @date 2022/12/23 13:47
 */
public class BackTrack_01_Combinations {

    public static void main(String[] args) {
        List<List<Integer>> res = combine(5, 4);
        System.out.println("res = " + res);
    }


    /**
     * 时间
     * 1 ms
     * 击败
     * 99.99%
     * 内存
     * 42.6 MB
     * 击败
     * 72.34%
     *
     * @param n
     * @param k
     * @return
     */
    private static List<List<Integer>> combine(int n, int k) {
        List<List<Integer>> res = new ArrayList<>();
        LinkedList<Integer> path = new LinkedList<>();
        combineHelper(n, k, 1, res, path);
        return res;
    }

    private static void combineHelper(int n, int k, int startIndex,
                                      List<List<Integer>> res, LinkedList<Integer> path) {
        if (path.size() == k) {
            res.add(new ArrayList<>(path));
            return;
        }
        int max = n - (k - path.size()) + 1;
        for (int i = startIndex; i <= max; i++) {
            path.add(i);
            combineHelper(n, k, i + 1, res, path);
            //取出路劲中的最后一个数据
            path.removeLast();
        }

    }
}
